package com.fxu.gen;

import com.fxu.gen.enums.TplEnum;
import com.fxu.tpl.entity.Table;

import java.io.File;

/**
 * Mapper数据库访问层
 */
public final class Mapper extends Base {
	public Mapper(Table table, File destFile) {
		super(table, TplEnum.Mapper, destFile);
	}
	
	@Override
    public String doReplace() {
		String fileTxt = super.doReplace();
		String module = table.getModule().getName();
		if (module.contains("/")) {
			module = module.substring(0, module.indexOf("/"));
		}
		fileTxt = fileTxt.replace("@DS(\"master_gen\")", "@DS(\"master_"+module+"\")");
//		if (table.getIsLog()) {
//			String antActLog = "@ActLog(insert=true, update=true, delete=true)";
//			if (AutoUtil.hasColumn(table.getColumnList(), Constant.CLN_CREATE_BY) && AutoUtil.hasColumn(table.getColumnList(), Constant.CLN_CREATE_TIME)) {
//				antActLog = "@ActLog(update=true, delete=true)";
//			}
//			int start = fileTxt.indexOf("@Mapper") + "@Mapper".length();
//			int end = fileTxt.indexOf("public interface");
//			if (start > 0 && end > start) {
//				fileTxt.replace(fileTxt.substring(start, end), antActLog);
//			} else {
//				logger.error("{}，{}标签有误", "@Mapper", "public interface");
//			}
//		}
		return fileTxt;
	}
}
